package vip.specimen.system.common.component;

import org.apache.shiro.authz.annotation.RequiresPermissions;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Component;
import vip.specimen.system.common.annotation.SystemJurisdiction;
import vip.specimen.system.service.SystemLogService;

import javax.annotation.Resource;
import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.Map;


/**
 * *@Author: 余浪人
 * *@Date：2021/3/15 17:14
 * *@Email:yulangren520@Gmail.com
 * *@QQ:1368419640
 **/
@Slf4j
@Aspect
@Component
public class SystemAspectComponent {
    @Resource
    private SystemLogService logService;

    /**
     * Controller层切点,注解方式
     */
    @Pointcut("@annotation(org.apache.shiro.authz.annotation.RequiresPermissions)")
    public void controllerAspect() throws Throwable {

    }

    /**
     * 前置通知 (在方法执行之前返回)用于拦截Controller层记录用户的操作的开始时间
     *
     * @param joinPoint 切点
     * @throws InterruptedException
     */
    @Before("controllerAspect()")
    public void doBefore(JoinPoint joinPoint) throws InterruptedException {
        //线程绑定变量（该数据只有当前请求的线程可见）
    }

    /**
     * 后置通知(在方法执行之后并返回数据) 用于拦截Controller层无异常的操作
     *
     * @param joinPoint 切点
     */
    @AfterReturning("controllerAspect()")
    public void after(JoinPoint joinPoint) {
        try {
            Map<String, String> map = getControllerMethodInfo(joinPoint);
            String permissionName = map.get("permissionName");
            this.logService.write(permissionName, joinPoint); // 调用写日志
        } catch (Exception e) {
            log.error("AOP后置通知异常", e);
        }
    }


    /**
     * 获取注解中对方法的描述信息 用于Controller层注解
     *
     * @param joinPoint 切点
     * @return 方法描述
     */
    private static Map<String, String> getControllerMethodInfo(JoinPoint joinPoint) throws Exception {
        Map<String, String> map = new HashMap<>();
        String targetName = joinPoint.getTarget().getClass().getName();//获取目标类名
        String methodName = joinPoint.getSignature().getName();//获取方法名
        Object[] arguments = joinPoint.getArgs();//获取相关参数
        Class targetClass = Class.forName(targetName); //生成类对象
        Method[] methods = targetClass.getMethods();//获取该类中的方法
        for (Method method : methods) {
            if (!method.getName().equals(methodName)) {
                continue;
            }
            Class[] cls = method.getParameterTypes();
            if (cls.length != arguments.length) { //比较方法中参数个数与从切点中获取的参数个数是否相同，原因是方法可以重载哦
                continue;
            }
            map.put("permissionName", method.getAnnotation(SystemJurisdiction.class).permissionName());
            String[] s = method.getAnnotation(RequiresPermissions.class).value();
            for (String s1 : s) {
                map.put("permission", s1);
            }
        }
        return map;
    }


}
